BitSparse: Compact Activation Storage for ReLU Feed-Forward Networks
BitSparse：ReLU 前馈网络的紧凑激活存储
🧭 训练显存优化 · 稀疏激活无损压缩
University of Cambridge · University of Oxford
提出 GPU 端无损稀疏激活压缩格式 BitSparse，在 Modded-NanoGPT 和 Nemotron-H 上降低每 token VRAM 31%/30%，开销不超过 1%
📌 工作参考价值（4 点）
- 若广告/推荐训练模型使用 ReLU/ReLU² 激活且前馈层存在大量稀疏零激活，可用 BitSparse 无损压缩激活缓存，降低训练 VRAM，支撑更大 batch 或更长序列，不牺牲精度。
- 压缩/解压以 Triton kernel 融合进 autograd function，开销 <1%，可作为自定义 activation checkpointing 的轻量替代或补充，适合在大模型训练框架中集成。
- 支持 BF16 和 FP8 激活，与混合精度训练兼容；业务上若已使用 FP8，可直接复用该压缩思路。
- 预分配 buffer 避免 variable size tensor 分配时的 host synchronization，这个工程技巧对推荐模型中动态序列长度样本的训练同样有价值。